#include "BlockQueue.hpp"
#include "Thread.hpp"
#include <vector>
#include <string>
#include <unistd.h>
#include "Task.hpp"
#include <ctime>
using namespace std;
using namespace ThreadModule;
using blockqueue_t = BlockQueue<Task>;
void PrintHello()
{
    cout << "hello world" << endl;
}
// int a = 10;
// 消费者线程执行方法
void Consumer(blockqueue_t &bq)
{
    // 消费者获取数据
    while (true)
    {
        sleep(1);
        // 1.消费者从blockqueue阻塞队列中取得任务
        Task t;
        bq.Pop(&t);
        // 2.执行任务
        // t.Excute();
        t();
        // cout << "消费者获取到的数据为:" << t.ResultToString() << endl;
    }
}
// 生产者线程执行方法
void Productor(blockqueue_t &bq)
{
    // 生产者生产数据
    srand(time(nullptr) ^ pthread_self());
    while (true)
    {
        sleep(1);
        // 1.获取任务
        // int a = rand() % 10 + 1;
        // usleep(1000);
        // int b = rand() % 20 + 1;
        // Task t(a, b);
        Task t = PrintHello;
        // 2.将获取到的任务放入到blockqueue阻塞队列中
        bq.Enqueue(t);
        // cout << "生产者生产的数据为:" << t.DebugToString() << endl;
    }
}
// 公共方法
void StartComm(vector<Thread<blockqueue_t>> *threads, int num, blockqueue_t &bq, func_t<blockqueue_t> func)
{
    for (int i = 0; i < num; i++)
    {
        // 为每个消费者线程构建名称
        string name = "thread-" + to_string(i + 1);
        // 管理每个消费者线程
        threads->emplace_back(func, bq, name);
        // 启动每个消费者线程
        (*threads)[threads->size() - 1].Start();
        // threads->back().Start();
    }
}
// 创建-管理消费者线程
void StartConsumer(vector<Thread<blockqueue_t>> *threads, int num, blockqueue_t &bq)
{
    StartComm(threads, num, bq, Consumer);
}
// 创建-管理生产者线程
void StartProductor(vector<Thread<blockqueue_t>> *threads, int num, blockqueue_t &bq)
{
    StartComm(threads, num, bq, Productor);
}
// 等待回收所有线程
void WaitALLThread(vector<Thread<blockqueue_t>> &threads)
{
    for (auto &thread : threads)
    {
        thread.Join();
    }
}
int main()
{
    // 阻塞队列
    blockqueue_t *bq = new blockqueue_t(5);
    // 管理生产者和消费者线程的数据结构
    vector<Thread<blockqueue_t>> threads;
    // 启动指定数量的消费者线程
    StartConsumer(&threads, 1, *bq);
    // 启动指定数量的生产者线程
    StartProductor(&threads, 1, *bq);
    // 等待回收所有线程
    WaitALLThread(threads);
    return 0;
}